package healthcheck

import (
	"context"
	"fmt"

	"github.com/containers/podman/v6/cmd/podman/common"
	"github.com/containers/podman/v6/cmd/podman/registry"
	"github.com/containers/podman/v6/libpod/define"
	"github.com/containers/podman/v6/pkg/domain/entities"
	"github.com/spf13/cobra"
)

var runCmd = &cobra.Command{
	Use:               "run CONTAINER",
	Short:             "Run the health check of a container",
	Long:              "Run the health check of a container",
	Example:           `podman healthcheck run mywebapp`,
	RunE:              run,
	Args:              cobra.ExactArgs(1),
	ValidArgsFunction: common.AutocompleteContainersRunning,
}

func init() {
	registry.Commands = append(registry.Commands, registry.CliCommand{
		Command: runCmd,
		Parent:  healthCmd,
	})
}

func run(_ *cobra.Command, args []string) error {
	response, err := registry.ContainerEngine().HealthCheckRun(context.Background(), args[0], entities.HealthCheckOptions{})
	if err != nil {
		return err
	}
	switch response.Status {
	case define.HealthCheckUnhealthy, define.HealthCheckStarting, define.HealthCheckStopped:
		registry.SetExitCode(1)
		fmt.Println(response.Status)
	}
	return err
}
